from sklearn import tree

X = [[0,0],[0,1]]
Y = [0,1]

print(type(Y))